/**
 * Copyright (c) Huawei Technologies Co., Ltd. 2023-2024. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/*!
 * \file flash_attention_score_grad_tiling_s1s2_bn2gs1s2_def.h
 * \brief
 */

#pragma once

#include <cstdint>
#include <register/tilingdata_base.h>
#include <tiling/tiling_api.h>

namespace optiling {

/////////////////////////////////////////////////////////////////////////
// S1S2_BNGS1S2
/////////////////////////////////////////////////////////////////////////
BEGIN_TILING_DATA_DEF(FlashAttentionScoreGradS1S2BNGS1S2BaseParams)
TILING_DATA_FIELD_DEF(uint32_t, coreNum);
TILING_DATA_FIELD_DEF(uint32_t, b);
TILING_DATA_FIELD_DEF(uint32_t, n2);
TILING_DATA_FIELD_DEF(uint32_t, g);
TILING_DATA_FIELD_DEF(uint32_t, s1);
TILING_DATA_FIELD_DEF(uint32_t, s2);
TILING_DATA_FIELD_DEF(uint32_t, d);
TILING_DATA_FIELD_DEF(float, scaleValue);
TILING_DATA_FIELD_DEF(float, keepProb);
TILING_DATA_FIELD_DEF(uint32_t, pseOptional);
TILING_DATA_FIELD_DEF(uint32_t, pseType);
TILING_DATA_FIELD_DEF(uint32_t, pseShapeType);
TILING_DATA_FIELD_DEF(uint32_t, pseDtype);
TILING_DATA_FIELD_DEF(uint32_t, attenMaskOptional);
TILING_DATA_FIELD_DEF(uint32_t, attenMaskDtype);
TILING_DATA_FIELD_DEF(uint32_t, attenMaskShapeType);
TILING_DATA_FIELD_DEF(int32_t, s1Token);  // pre_tokens
TILING_DATA_FIELD_DEF(int32_t, s2Token);  // next_tokens
TILING_DATA_FIELD_DEF(uint32_t, sparseMode);
TILING_DATA_FIELD_DEF(uint32_t, isSparse);
TILING_DATA_FIELD_DEF_ARR(uint32_t, 64, prefixN);
TILING_DATA_FIELD_DEF(uint32_t, attenMaskCompressMode);
TILING_DATA_FIELD_DEF(uint32_t, attenMaskS2Size);
END_TILING_DATA_DEF;
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGradS1S2BNGS1S2BaseParamsOp, FlashAttentionScoreGradS1S2BNGS1S2BaseParams)

BEGIN_TILING_DATA_DEF(FlashAttentionScoreGradS1S2BNGS1S2SplitCoreParams)
TILING_DATA_FIELD_DEF(uint32_t, s1Outer);
TILING_DATA_FIELD_DEF(uint32_t, s1CvRatio);
TILING_DATA_FIELD_DEF(uint32_t, s1Inner);
TILING_DATA_FIELD_DEF(uint32_t, s1CvInner);
TILING_DATA_FIELD_DEF(uint32_t, s1Tail);
TILING_DATA_FIELD_DEF(uint32_t, s1CvTail);
TILING_DATA_FIELD_DEF(uint32_t, s2Outer);
TILING_DATA_FIELD_DEF(uint32_t, s2CvRatio);
TILING_DATA_FIELD_DEF(uint32_t, s2Inner);
TILING_DATA_FIELD_DEF(uint32_t, s2Tail);
TILING_DATA_FIELD_DEF(uint32_t, baseMN);
TILING_DATA_FIELD_DEF(uint32_t, sfmgdOuter);
TILING_DATA_FIELD_DEF(uint32_t, sfmgdFactor);
TILING_DATA_FIELD_DEF(uint32_t, sfmgdTail);
TILING_DATA_FIELD_DEF(uint32_t, blockOuter);
TILING_DATA_FIELD_DEF(uint32_t, bandIdx);
END_TILING_DATA_DEF;
// 固定写法不能换行，会失败
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGradS1S2BNGS1S2SplitCoreParamsOp, FlashAttentionScoreGradS1S2BNGS1S2SplitCoreParams)

BEGIN_TILING_DATA_DEF(BlockNumListParams)
TILING_DATA_FIELD_DEF_ARR(uint32_t, 50, blockStarts);
TILING_DATA_FIELD_DEF_ARR(uint32_t, 50, blockEnds);
END_TILING_DATA_DEF;
REGISTER_TILING_DATA_CLASS(BlockNumListParamsOp, BlockNumListParams)

BEGIN_TILING_DATA_DEF(FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
TILING_DATA_FIELD_DEF_STRUCT(FlashAttentionScoreGradS1S2BNGS1S2BaseParams, s1s2BNGS1S2BaseParams);
TILING_DATA_FIELD_DEF_STRUCT(FlashAttentionScoreGradS1S2BNGS1S2SplitCoreParams, s1s2BNGS1S2SplitCoreParams);
TILING_DATA_FIELD_DEF_STRUCT(TCubeTiling, mm1TilingData);
TILING_DATA_FIELD_DEF_STRUCT(TCubeTiling, mm2TilingData);
TILING_DATA_FIELD_DEF_STRUCT(TCubeTiling, mm3TilingData);
TILING_DATA_FIELD_DEF_STRUCT(SoftMaxTiling, softmaxTilingData);
TILING_DATA_FIELD_DEF_STRUCT(SoftMaxTiling, softmaxGradTilingData);
TILING_DATA_FIELD_DEF_STRUCT(BlockNumListParams, s1s2BNGS1S2BlockNumList);
TILING_DATA_FIELD_DEF_STRUCT(PreParams, preTilingData);
TILING_DATA_FIELD_DEF_STRUCT(PostParams, postTilingData);
END_TILING_DATA_DEF;
// 固定写法不能换行，会失败
// BSND 1000000xxxxxxxx0x434(5)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000000002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000000003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000111003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000111002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000011003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000011002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000101003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000101002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000001003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000001002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000110003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000110002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000010003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000010002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000100003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000100002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001000002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001000003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001111003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001111002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001011003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001011002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001101003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001101002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001001003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001001002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001110003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001110002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001010003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001010002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001100003434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001100002434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
// SBND  1000000xxxxxxxx1x434(5)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000000012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000000013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000111013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000111012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000011013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000011012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000101013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000101012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000001013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000001012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000110013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000110012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000010013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000010012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000100013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000100012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001000012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001000013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001111013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001111012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001011013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001011012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001101013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001101012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001001013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001001012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001110013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001110012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001010013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001010012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001100013434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001100012434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
// BNSD  1000000xxxxxxxx2x434(5)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000000022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000000023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000111023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000111022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000011023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000011022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000101023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000101022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000001023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000001022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000110023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000110022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000010023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000010022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000100023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000100022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001000022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001000023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001111023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001111022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001011023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001011022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001101023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001101022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001001023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001001022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001110023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001110022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001010023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001010022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001100023434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001100022434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
// TND  1000000xxxxxxxx3x434(5)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000000032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000000033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000111033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000111032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000011033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000011032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000101033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000101032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000001033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000001032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000110033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000110032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000010033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000010032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000100033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000000100032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001000032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001000033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001111033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001111032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001011033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001011032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001101033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001101032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001001033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001001032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001110033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001110032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001010033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001010032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001100033434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
REGISTER_TILING_DATA_CLASS(FlashAttentionScoreGrad_10000000001100032434, FlashAttentionScoreGradTilingDataUs1s2Bbn2gs1s2)
}  // namespace optiling
